package com.hlx.entiry.vo;

import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.ai.chat.messages.Message;

@NoArgsConstructor
@Data
public class MessageVO {
	private String role;
	private String content;

	public MessageVO(Message message) {
		switch (message.getMessageType()) {
			case USER -> this.role = "user";
			case ASSISTANT -> this.role = "assistant";
		}
		this.content = message.getText();
	}
}
